package JackSu.aobing;

import java.util.HashMap;
import java.util.Map;

/**
 * LRUTest
 *
 * @author Syzer
 * @Description LRU算法
 * @Date 2021/7/20 10:12
 */
public class LRUTest2 {

    class Node {
        int key;
        int val;
        Node pre;
        Node next;

        public Node() {
        }

        public Node(int key, int val) {
            this.key = key;
            this.val = val;
        }
    }

    Node head;
    Node tail;
    int capacity;
    int size;
    Map<Integer, Node> map;

    public LRUTest2(int capacity) {
        this.size = 0;
        this.capacity = capacity;
        this.map = new HashMap<>();
        this.head = new Node();
        this.tail = new Node();
        this.head.next = tail;
        this.tail.pre = head;
    }

    public int get(int key) {
        // 获取
        Node node = map.get(key);
        if(node == null) {
            return -1;
        }
        // 放到最新（头部）
        moveToHead(node);
        return node.val;
    }

    public void put(int key, int value) {
        Node node = map.get(key);
        if(node == null){
            // 新增
            node = new Node(key, value);
            addToHead(node);
            map.put(key, node);
            size++;
            // 丢弃最早的
            if (size > capacity) {
                Node removeNode = removeTail();
                map.remove(removeNode.key);
                size--;
            }
        } else {
            // 修改
            node.val = value;
            moveToHead(node);
        }
    }

    private void addToHead(Node node) {
        node.next = head.next;
        node.pre = head;
        head.next.pre = node;
        head.next = node;
    }

    private void removeNode(Node node) {
        node.pre.next = node.next;
        node.next.pre = node.pre;
    }

    private void moveToHead(Node node) {
        removeNode(node);
        addToHead(node);
    }

    private Node removeTail() {
        Node removeNode = tail.pre;
        removeNode(removeNode);
        return removeNode;
    }

    public static void main(String[] args) {
        LRUTest2 lRUCache = new LRUTest2(2);
//        lRUCache.put(1, 1); // 缓存是 {1=1}
//        lRUCache.put(2, 2); // 缓存是 {1=1, 2=2}
//        lRUCache.put(3, 3); // 缓存是 {1=1, 2=2}
//        lRUCache.put(4, 4); // 缓存是 {1=1, 2=2}
//        lRUCache.get(4);    // 返回 1
//        lRUCache.get(3);    // 返回 1
//        lRUCache.get(2);    // 返回 1
//        lRUCache.get(1);    // 返回 1
//        lRUCache.put(5, 5); // 该操作会使得关键字 2 作废，缓存是 {1=1, 3=3}
//        lRUCache.get(1);    // 返回 -1 (未找到)
//        lRUCache.get(2);    // 返回 -1 (未找到)
//        lRUCache.get(3);    // 返回 -1 (未找到)
//        lRUCache.get(4);    // 返回 -1 (未找到)
//        lRUCache.get(5);    // 返回 4
        lRUCache.put(2, 1);
        lRUCache.put(2, 2);
        lRUCache.get(2);
        lRUCache.put(1, 1);
        lRUCache.put(4, 1);
        lRUCache.get(2);
    }
}
